'''
bge-m3 文本向量模型使用
两个方法：
1.文本字符串向量化 emb.embed_query(text)
2.文本字符串向量化 emb_embed_document(text_list)
'''

from langchain_ollama import OllamaEmbeddings

emb = OllamaEmbeddings(
    model="bge-m3:latest",
    base_url="http://localhost:64342"
)

print(emb)

#测试词输入
text = """
姓名：马云（英文名：Jack Ma）
出生日期：1964年9月10日
出生地：中国浙江省杭州市
毕业院校：杭州师范大学（原杭州师范学院），英语专业
花名：风清扬（源自金庸小说《笑傲江湖》中的神秘高手风清扬）
"""

text_list = ["姓名：马云（英文名：Jack Ma）","出生日期：1964年9月10日","出生地：中国浙江省杭州市"]

vec_text = emb.embed_query(text)
print(vec_text)
print(len(vec_text))

vec_list = emb.embed_documents(text_list)
print(vec_list)
print(vec_list[0])